package appli;

import index.BernoulliIndex;
import index.Corpus;
import index.Tweet;

import java.util.HashMap;

/**
 * Main de la classification 
 * @author Marie, Nath
 *
 */
public class Classifieur {

	// dossier a analyser
	public static final String corpus = "ressources/train.txt";
	public static final String stopWordsFile = "ressources/antidico.txt";
	// dictionnaire des mots et leurs tdIdf
	public static HashMap<String, Float> dictionnaire;

	public static void main(String[] args) {

		String dossierTest = "ressources/test.txt";
		if(args.length >= 1)
			dossierTest = args[0];
		Corpus corpusTest = new Corpus(dossierTest, false);
		
		
		Corpus corpusApprentissage = new Corpus(corpus, false);
		// le meilleur analyseur
		BernoulliIndex analyser = new BernoulliIndex(corpusApprentissage);
		
		// on parcourt tous les tweets
		for (Tweet t : corpusTest.getDocuments()) {
			System.out.println(analyser.type(t));
		}
/*
 * String fichierSortie = "ressources/MASRI_NIVET-test-EIT-12.txt";
		
		FileWriter writer = null;
		try {
			writer = new FileWriter(fichierSortie);
			for (Tweet t : corpusTest.getDocuments()) {
				writer.write(analyser.type(t) + "\n");
				System.out.println(analyser.type(t));
			}
			writer.close();
		} catch (IOException e) {
			e.printStackTrace();
		}
		*/
	}

}
